package com.zqj.stream1;

import com.zqj.lambda2.Employee;
import com.zqj.lambda2.EmployeeData;
import org.junit.Test;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.stream.Stream;

/**
 * 测试Stream的中间操作
 */
public class StreamAPITest1 {

    //1.筛选与切片

    /**
     * filter(Predicate p)-接收Lambda，从流中排除某些元素。
     * limit(n)-截断流，使其元素不超过给定数量
     * skip(n)-跳过元素，返回一个扔掉前n个元素的流。若流中的元素不足n个，则返回一个空流和limit()互补
     * distinct()-筛选，通过流所生成的元素的hashCode()和equals()去除重复元素
     */
    @Test
    public void test1(){
        // filter(Predicate p)-接收Lambda，从流中排除某些元素。
        List<Employee> list = EmployeeData.getEmployees();
        Stream<Employee> stream = list.stream();
        //查询工资不超过七千的
        stream.filter(e -> e.getSalary() > 7000).forEach(System.out::println);
        System.out.println("****************");
        //limit(n)-截断流，使其元素不超过给定数量
        list.stream().limit(3).forEach(System.out::println);

        System.out.println("***************");
//        skip(n)-跳过元素，返回一个扔掉前n个元素的流。若流中的元素不足n个，则返回一个空流和limit()互补
        list.stream().skip(3).forEach(System.out::println);
        System.out.println("*****************");
        //distinct()-筛选，通过流所生成的元素的hashCode()和equals()去除重复元素

        //要重写hashCode和equals()方法
        list.stream().distinct().forEach(System.out::println);

    }
    //映射
    @Test
    public void test2(){
        /**
         * map(Function f)-接收一个函数作为参数，将元素转换成其他形式或提取信息，该函数会被
         * 应用到每个元素上，并将其映射成一个新的元素(类似List里的add())
         */
        List<String> list = Arrays.asList("aa","bb","cc","dd");
        list.stream().map(str ->str.toUpperCase()).forEach(System.out :: println);

        System.out.println("***************************************");
        //练习:获取员工姓名长度大于3的员工的姓名
        List<Employee> employees = EmployeeData.getEmployees();
        Stream<String> nameStream = employees.stream().map(Employee::getName);
        nameStream.filter(name->name.length() >3).forEach(System.out::println);
        //练习2.
        Stream<Stream<Character>> streamStream = list.stream().map(StreamAPITest1::fromStringToStream);
        streamStream.forEach(s->{
            s.forEach(System.out::println);
        });
        System.out.println("***************************************************");
        //flatMap(Function f)-接收一个函数作为参数，将流中的每一个值转换成另一个流，然后把所有
        //流连接成一个流 （类似List里面的 addAll()）
        Stream<Character> characterStream = list.stream().flatMap(StreamAPITest1::fromStringToStream);
        characterStream.forEach(System.out::println);

    }
    //将字符串中的多个字符构成的集合转换为对应的Stream的实例
    public static Stream<Character> fromStringToStream(String str){
        ArrayList<Character> list = new ArrayList<>();

        for (Character c: str.toCharArray()) {
            list.add(c);
        }
        return list.stream();
    }

    //3-排序
    @Test
    public void test4(){
        // sorted()——自然排序
        List<Integer> list = Arrays.asList(20, 12, 57, -1, 0,3, 78, 100, 10);
        list.stream().sorted().forEach(System.out::println);

        //抛异常了，Employee没有实现Comparable接口
//        List<Employee> employees = EmployeeData.getEmployees();
//        employees.stream().sorted().forEach(System.out::println);

        System.out.println("*******************************************");
        // sorted(Comparator com)——定制排序
        List<Employee> employees = EmployeeData.getEmployees();
//        employees.stream().sorted((e1,e2)->Integer.compare(e1.getAge(),e2.getAge()))
//                .forEach(System.out::println);
        //先按年龄升序再按工资倒叙
        employees.stream().sorted((e1,e2)->{
            int ageValue = Integer.compare(e1.getAge(),e2.getAge());
            if (ageValue !=0){
                return ageValue;
            }else {
                return -Double.compare(e1.getSalary(),e2.getSalary());
            }
                })
                .forEach(System.out::println);


    }

}
